#!/bin/bash

# dataset_name=('semval-RE')
# dataset_name=('conll04')
# aug_rules=('add_triple' 'add_same_type_triple' 'remove_triple' 'replace_same_type_entity' 'swap_triple')
# timestamp=$(date +%s)
# echo $@
# echo $#

dataset_name=($1)
timestamp=$2
aug_timies=$3
few_shot_num=($4)

shift 4

aug_rules=$@
# echo $aug_rules

# for dataset_name in "${dataset_name[@]}"; do
#     python ./data_augmentation/generate_data.py \
#         --path_to_model_config="./configs/gen_data_config.yaml" \
#         --dataset_name=$dataset_name \
#         --rules="${aug_rules[@]}"
# done

is_few_shot=0

# for dataset in "${dataset_name[@]}"; do
#     python ./data_augmentation/generate_data.py \
#         --dataset_name="$dataset" \
#         --save_dir=$timestamp \
#         --aug_times=$aug_timies \
#         --rules="$aug_rules"

# done

python ./data_augmentation/generate_with_llm.py \
    --dataset_name="$dataset_name" \
    --save_dir=$timestamp \
    --aug_times=$aug_timies \
    --few_shot_num=$few_shot_num \
    --is_few_shot=$is_few_shot \
    --rules="$aug_rules"
